cfg = {
    'gpus': [0, 1, 2, 3],
    'cuda': True,
    'workers': 8,
    'output_dir': 'output/resnet50',
    'pretrain': '',
    'synce_bn': False,

    'dataset':{
        'name': 'iCartoonFace',
        'root':  './data/personai_icartoonface_dettrain/icartoonface_dettrain',
        'anno_root': './data/personai_icartoonface_dettrain/icartoonface_dettrain.csv',
        'image_size': [640, 640],
    },
    
    'model':{
        'name': 'resnet50',
        'num_classes': 1,
        'bias_value': -2.19,
        'pretrain': False,
        'fpn_channels': [256, 512, 1024],
        'return_layers': ['layer1', 'layer2', 'layer3'],
    },

    'train':{
        'batch_size_per_gpu': 12,
        'max_epoch': 140,
        'optimizer': 'adam',
        'lr': 5e-4,
        'lr_factor': 0.1,
        'lr_step': [90, 120],
        'wd': 1e-4,
        'moentum': 0.9,
        'heatmap_weight': 1.0,
        'wh_weight': 5.0,
    },

    'test':{
        'batch_size_per_gpu': 12,
    },
}